#!/usr/bin/env python

import redis

from ast import literal_eval




class RedisQueue(object):
    def __init__(self, name, redis_conf, namespace='queue'):
        pool = redis.ConnectionPool(host=redis_conf["host"],
                                    port=redis_conf["port"],
                                    db=redis_conf["db"],
                                    password=redis_conf["password"],
                                    decode_responses=True)
        self.__db = redis.Redis(connection_pool=pool)
        self.key = '%s:%s' %(namespace, name)

    def qsize(self):
        return self.__db.llen(self.key)  # 返回队列里面list内元素的数量

    def put(self, item):
        self.__db.rpush(self.key, item)  # 添加新元素到队列最右方

    def get_wait(self, timeout=None):
        # 返回队列第一个元素，如果为空则等待至有元素被加入队列（超时时间阈值为timeout，如果为None则一直等待）
        item = self.__db.blpop(self.key, timeout=timeout)
        # if item:
        #     item = item[1]  # 返回值为一个tuple
        return item

    def get_nowait(self):
        # 直接返回队列第一个元素，如果队列为空返回的是None
        item = self.__db.lpop(self.key)  
        return item

    # def get(self):
    #
    #     return self.__db.hkeys(1)

class RedisFunc(object):
    def __init__(self, dic_name, redis_conf):
        pool = redis.ConnectionPool(host=redis_conf["host"],
                                    port=redis_conf["port"],
                                    db=redis_conf["db"],
                                    password=redis_conf["password"],
                                    decode_responses=True)
        
        self.__db = r = redis.Redis(connection_pool=pool)
        self.dic_name = dic_name
        
    def get_single(self, key_name):
        value_data = self.__db.hget(self.dic_name, key_name)
        if value_data:
            unpacked_object = literal_eval(value_data)
            return unpacked_object
        else:
            return value_data
        
    def set_single(self, key_name, value_data):
        self.__db.hset(self.dic_name, key_name, value_data)
        
    
    def get_batches(self):
        key_list = self.__db.hkeys(self.dic_name)
        value_list = self.__db.hvals(self.dic_name)
        return key_list, value_list
        
    

    